
package cartagows.wsframework.wssecurity.policy.builders;

import java.util.Iterator;
import java.util.List;

import javax.xml.namespace.QName;

import org.apache.axiom.om.OMElement;
import org.apache.neethi.Assertion;
import org.apache.neethi.AssertionBuilderFactory;
import org.apache.neethi.Policy;
import org.apache.neethi.PolicyEngine;
import org.apache.neethi.builders.AssertionBuilder;
import org.apache.neethi.builders.xml.XmlPrimtiveAssertion;

import cartagows.wsframework.wssecurity.policy.SP11Constants;
import cartagows.wsframework.wssecurity.policy.SPConstants;
import cartagows.wsframework.wssecurity.policy.model.AlgorithmSuite;
import cartagows.wsframework.wssecurity.policy.model.AsymmetricBinding;
import cartagows.wsframework.wssecurity.policy.model.InitiatorToken;
import cartagows.wsframework.wssecurity.policy.model.Layout;
import cartagows.wsframework.wssecurity.policy.model.RecipientToken;

public class AsymmetricBindingBuilder implements AssertionBuilder {

	public Assertion build(OMElement element, AssertionBuilderFactory factory) throws IllegalArgumentException {

		AsymmetricBinding asymmetricBinding =  new AsymmetricBinding(SPConstants.SP_V11);

		Policy policy = PolicyEngine.getPolicy(element.getFirstElement());
		policy = (Policy) policy.normalize(false);

		for (Iterator iterator = policy.getAlternatives(); iterator.hasNext();) {
			processAlternative((List) iterator.next(), asymmetricBinding);

			 break;
		}

		return asymmetricBinding;
	}

	private void processAlternative(List assertions, AsymmetricBinding asymmetricBinding) {

		Assertion assertion;
		QName name;

		for (Iterator iterator = assertions.iterator(); iterator.hasNext();) {
			assertion = (Assertion) iterator.next();
			name = assertion.getName();

			if (SP11Constants.INITIATOR_TOKEN.equals(name)) {
				InitiatorTokenBuilder itb = new InitiatorTokenBuilder();
				assertion = itb.build(((XmlPrimtiveAssertion ) assertion).getValue(), null);

				asymmetricBinding.setInitiatorToken((InitiatorToken) assertion);

			} else if (SP11Constants.RECIPIENT_TOKEN.equals(name)){
				RecipientTokenBuilder rtb = new RecipientTokenBuilder();
				assertion = rtb.build(((XmlPrimtiveAssertion ) assertion).getValue(), null);
				
				asymmetricBinding.setRecipientToken((RecipientToken) assertion);

			} else if (SP11Constants.ALGORITHM_SUITE.equals(name)) {
				AlgorithmSuiteBuilder asb = new AlgorithmSuiteBuilder();
				assertion = asb.build(((XmlPrimtiveAssertion ) assertion).getValue(), null);
				
				asymmetricBinding.setAlgorithmSuite((AlgorithmSuite) assertion);

			} else if (SP11Constants.LAYOUT.equals(name)) {
				LayoutBuilder lb = new LayoutBuilder();
				assertion = lb.build(((XmlPrimtiveAssertion ) assertion).getValue(), null);
				
				asymmetricBinding.setLayout((Layout) assertion);

			} else if (SP11Constants.INCLUDE_TIMESTAMP.equals(name)) {
				asymmetricBinding.setIncludeTimestamp(true);

			} else if (SPConstants.ENCRYPT_BEFORE_SIGNING.equals(name.getLocalPart())) {
				asymmetricBinding.setProtectionOrder(SPConstants.ENCRYPT_BEFORE_SIGNING);

			} else if (SPConstants.SIGN_BEFORE_ENCRYPTING.equals(name.getLocalPart())) {
				asymmetricBinding.setProtectionOrder(SPConstants.SIGN_BEFORE_ENCRYPTING);

			} else if (SPConstants.ENCRYPT_SIGNATURE.equals(name.getLocalPart())) {
				asymmetricBinding.setSignatureProtection(true);

			} else if (SPConstants.PROTECT_TOKENS.equals(name.getLocalPart())) {
				asymmetricBinding.setTokenProtection(true);

			} else if (SPConstants.ONLY_SIGN_ENTIRE_HEADERS_AND_BODY.equals(name.getLocalPart())) {
				asymmetricBinding.setEntireHeadersAndBodySignatures(true);
			}
		}
	}

	public QName[] getKnownElements() {
		return new QName[]{SP11Constants.ASYMMETRIC_BINDING};
	}

}
